/*
 * systrace - Should be a tool to analyze system call traces but can be used
 *            to analyze sequences of generic strings
 *
 * Copyright (c) 2011 Matteo Cicuttin
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 * 3. The name of the author may not be used to endorse or promote products
 * derived from this software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <iostream>

#include "SubtraceAnalysis.h"

SubtraceAnalysis::SubtraceAnalysis()
{
	set_module_name("subtrace");
	_stlen = 6;
	_sl = 0;
}

void
SubtraceAnalysis::flush(void)
{
	_callmap.clear();
	_subtrace.clear();
	_sl = 0;
}

void
SubtraceAnalysis::add_sample(const string& s)
{
	list<string>::iterator itor;
	
	_subtrace.push_back(s);
	_sl++;
	
	if (_sl != _stlen)
		return;
	
	_oss.str("");
	itor = _subtrace.begin();
	for(int i = 0; i < _stlen; i++)
	{
		_oss << (*itor);
		if (i != _stlen - 1)
			_oss << "; ";
		itor++;
	}
	
	if ( _callmap.find(_oss.str()) == _callmap.end() )
		_callmap[_oss.str()] = 1;
	else
		_callmap[_oss.str()]++;
	
	_subtrace.pop_front();
	_sl--;
}

void
SubtraceAnalysis::print_result(ostream& os)
{
	map<string,int>::iterator itor;
	for(itor = _callmap.begin(); itor != _callmap.end(); itor++)
		os << (*itor).first <<  ": " << (*itor).second << endl;
	os << "Found " << _callmap.size() << " different sequences" << endl;
}

void
SubtraceAnalysis::set_subtrace_length(int len)
{
	_stlen = len;
}

int
SubtraceAnalysis::subtrace_length(void)
{
	return _stlen;
}
